import torch as pt
import numpy as np
from python_ai.common.xcommon import *

pt.set_printoptions(edgeitems=100)
np.random.seed(777)
pt.manual_seed(777)

sep('cat dim=1')
x1t = pt.arange(1, 5*2+1).view(5, 2)
x2t = pt.arange(1, 5*4+1).view(5, 4)
xt = pt.cat((x1t, x2t), dim=1)
print(xt)

sep('cat dim=2')
x3t = pt.arange(1, 2*6+1).view(2, 6)
xt = pt.cat((xt, x3t), dim=0)
print(xt)